# -*- coding: utf-8 -*-
# @Time : 2023/4/5 16:01
# @Author : 陈鹏飞
# @Email ： 2578925789@qq.com
# @File : export_onnx
# @Description : 

import os

if __name__ == "__main__":
    from ultralytics import YOLO

    model = YOLO('./runs/detect/train35/weights/best.pt')  # load a custom trained

    # Export the model
    model.export(format='onnx', imgsz=[192, 320], opset=12)


